Grid management in Python